package com.llmops.demo;

public class CodeTest {
//    public static void main(String[] args) {
//        WorkflowGraph workflow = new WorkflowGraph("1");
//        Context context = workflow.getContext();
//        context.setVariable("sessionId",3);
//
//        workflow.addNode(new UserInputNode("data_input",Collections.singletonMap("input","你是谁")));
//        String code = """
//                def main(a):
//                 return len(a);
//                """;
//        Map in = new HashMap();
//        in.put("code",code);
//        in.put("inputs", CollUtil.newArrayList("data_input_output"));
//        workflow.addNode(new CodeNode("code_A", in));
////        workflow.addNode(new LLMNode("llm_inferenceB", Collections.singletonMap("model", "gpt-B")));
//
//
//        workflow.addNode(new OutputNode("result_exporterA", Collections.singletonMap("v", "A")));
////        workflow.addNode(new OutputNode("result_exporterB", Collections.singletonMap("v", "B")));
//
//        workflow.addEdge(new Edge("data_input", "code_A"));
////        workflow.addEdge(new Edge("data_loader", "llm_inferenceB", Edge.ConditionType.EQUAL,"B"));
//
//        workflow.addEdge(new Edge("code_A", "result_exporterA"));
////        workflow.addEdge(new Edge("llm_inferenceB", "result_exporterB"));
//
//        workflow.validate();
//        workflow.execute();
//
////        workflow.execute(context);
//
//        System.out.println("Workflow execution completed");
//    }
}